#!/usr/bin/env python2
# -*- coding:utf-8 -*-

import unittest
import os
from ssh_key_fp import KeyManager
from ssh_key_fp import LoginParser


dir = os.path.dirname(__file__)

class TestKeyManager(unittest.TestCase):
  def test_load(self):
    km = KeyManager("{}/../priv/staging-vos100_authorized_keys".format(dir))
    km.load_authorized_keys()
    km.puts() 
    self.assertEqual(1, 1)

class TestLoginParser(unittest.TestCase):
  def test_parse(self):
    km = KeyManager("{}/../priv/staging-vos100_authorized_keys".format(dir))
    km.load_authorized_keys()
    LoginParser(km, "{}/../priv/staging-vos100_audit.log".format(dir))
    LoginParser(km, "{}/../priv/staging-vos100_secure".format(dir))

if __name__ == '__main__':
  unittest.main()
